'use strict';

module.exports = app => {
  const { STRING, INTEGER } = app.Sequelize;

  const Train = app.model.define('train', {
    id: {
      autoIncrement: true,
      primaryKey: true,
      type: INTEGER,
    },
    train_num: STRING,
    train_type: STRING,
  });

  return Train;
};
